package com.snopy.leetcode.index1_1000.index201_300;

import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

/**
 * @author snopy
 * @version 1.0.0
 * @classname Question219
 * @description 存在重复元素II
 * 给你一个整数数组nums 和一个整数k ，判断数组中是否存在两个 不同的索引i和j ，
 * 满足 nums[i] == nums[j] 且 abs(i - j) <= k 。如果存在，返回 true ；否则，返回 false 。
 * 示例1：
 * 输入：nums = [1,2,3,1], k = 3
 * 输出：true
 * 示例 2：
 *
 * 输入：nums = [1,0,1,1], k = 1
 * 输出：true
 * 示例 3：
 *
 * 输入：nums = [1,2,3,1,2,3], k = 2
 * 输出：false
 * @email 77912204@qq.com
 * @date 2022/04/07 21:18
 */
public class Question219 {
    public static void main(String[] args) {
        final boolean b = containsNearbyDuplicate(new int[]{1, 2, 3, 1, 1}, 2);
        System.out.println(b);
    }
    public static boolean containsNearbyDuplicate(int[] nums, int k) {
        if (nums.length < 2 || k == 0){
            return false;
        }
        Map<Integer,Integer> map = new HashMap<>();
        for (int i=0;i<nums.length;i++){
            if (map.containsKey(nums[i])){
                if (Math.abs(map.get(nums[i])-i)<=k){
                    return true;
                }
            }
            map.put(nums[i],i);
        }

        return true;
    }
    /**
     * @Description 滑动窗口解法 ，集合中最多有k个元素
     * @param nums:
     * @param k:
     * @return: boolean
     * @Date 2022-04-07
     * @Author txl77912204@gmail.com
     **/
    public static boolean containsNearbyDuplicate2(int[]nums,int k){
        Set<Integer> set = new HashSet<Integer>();
        int length = nums.length;
        for (int i = 0; i < length; i++) {
            if (i > k) {
                set.remove(nums[i - k - 1]);
            }
            if (!set.add(nums[i])) {
                return true;
            }
        }
        return false;
    }
}
